#/*************************************************************************
#    > File Name: print_param.py
#    > Author: Yan Wang
#    > Mail: wangyan@imnu.edu.cn
#    > Created Time: 2022年04月10日 星期日 23时28分13秒
# ************************************************************************/
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
sys.path.append('./Lib/')
from File_Operation import *


np.set_printoptions(threshold=np.inf)
pd.set_option('display.max_columns',None)
pd.set_option('display.max_rows',None)

df = pd.read_table("../input_data/data/data_grid_sba_tanb.tsv")
begin=0
end=len(df)
target_folder="input_param"

def Print_MG_Xection():
	label="#  Integrated weight (pb)  :"
	xection_word_list=Extract_One_Sentence_From_MultiFile(target_folder,label)
	cross=[eval(word.split(":",2)[1]) for word in xection_word_list]
	name=["run_{}".format(num) for num in range(end-begin+1)]
	tb=df["tanb"][begin:end].to_list() 
	sba=df["sba"][begin:end].to_list() 
	mh=df["mh"][begin:end].to_list() 
	mH=df["mH"][begin:end].to_list() 
	mA=df["mA"][begin:end].to_list() 
	mch=df["mHc"][begin:end].to_list() 
	para=[name,tb,sba,mh,mH,mA,mch,cross]
	para=list(map(list,zip(*para)))
	out_file="../output_data/MG_xection/MG_xection_sba_tb.dat"
	f=open(out_file,'w')
	f.write("#run_name            frblock#1            frblock#2            mass#25              mass#35              mass#36              mass#37              cross")
	f.write("\n")
	print(len(df))
	for i in range(len(para)):
		f.write("            ".join(map(str,para[i])))
		f.write("\n")
	f.close()

Print_MG_Xection()
